package org.ytu.classifier.kmeans;

import java.util.ArrayList;
import java.util.List;

public class Cluster {
   private String name;
   private Centroid centroid;
   private List<FeaturedData> dataList = new ArrayList<FeaturedData>();

   public Cluster(String name) {
      this.name = name;
   }

   public void addElementToCluster(FeaturedData data) {
      this.dataList.add(data);
   }

   public String getName() {
      return name;
   }

   public Centroid getCentroid() {
      return centroid;
   }

   public void setCentroid(Centroid centroid) {
      this.centroid = centroid;
   }

   public List<FeaturedData> getDataList() {
      return dataList;
   }

   public void setDataList(List<FeaturedData> dataList) {
      this.dataList = dataList;
   }
}
